#![no_std]
#![crate_type = "staticlib"]
use cuda_std::*;

// 35KB per array, 3 arrays = 105KB total (well above 64KB limit)
const ARRAY_SIZE: usize = 35 * 1024 / 4; 

static BIG_ARRAY_1: [u32; ARRAY_SIZE] = [111u32; ARRAY_SIZE];
static BIG_ARRAY_2: [u32; ARRAY_SIZE] = [222u32; ARRAY_SIZE];
static BIG_ARRAY_3: [u32; ARRAY_SIZE] = [333u32; ARRAY_SIZE];

pub unsafe fn test_kernel(out: *mut u32) {
    *out = BIG_ARRAY_1[0] + BIG_ARRAY_2[0] + BIG_ARRAY_3[0];
}

use std::path::PathBuf;

fn main() {
    let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap());
    let out_path = PathBuf::from(std::env::var("OUT_DIR").unwrap());

    println!("cargo::warning=Building GPU kernels with constant memory ENABLED");
    println!("cargo::warning=This should trigger constant memory overspill errors");
    
    cuda_builder::CudaBuilder::new(manifest_dir.join("kernels"))
        .copy_to(out_path.join("kernels.ptx"))
        // Enable constant memory placement to trigger the overspill error
        .use_constant_memory_space(true)
        .build()
        .unwrap();
}

[workspace]

[package]
name = "const_memory_overflow"
version = "0.1.0"
edition = "2021"

[dependencies]
cust = { path = "../../../crates/cust" }

[build-dependencies]
cuda_builder = { path = "../../../crates/cuda_builder", features = ["rustc_codegen_nvvm"] }

[workspace]

[package]
name = "const_memory_overflow_kernels"
version = "0.1.0"
edition = "2021"

[dependencies]
cuda_std = { path = "../../../../crates/cuda_std" }

[lib]
crate-type = ["cdylib", "rlib"]

use cust::prelude::*;

fn main() {  
    println!("This example is meant to fail at compile time!");
    println!("It demonstrates the improved error messages for constant memory overflow.");
}

static BIG_ARRAY_1: [u32; ARRAY_SIZE] = [111u32; ARRAY_SIZE];
#[cuda_std::address_space(global)]
static BIG_ARRAY_2: [u32; ARRAY_SIZE] = [222u32; ARRAY_SIZE];

.use_constant_memory_space(false)